model_name = "Reinforce"
dataset_name = "tmall"
seed = 42  # 设定一个随机种子
n_users = 512

BATCH_SIZE = 512
EPOCHS = 10
LEARNING_RATE = 1e-4
MOMENTUM = 0.9
WEIGHT_DECAY = 1e-5
heads=8
group_num = 100
interest_num=4
short_time=20
if dataset_name == "xlong":
    total_behavior_num=1000
    id_embedding_dim=16
elif dataset_name == "taobao":
    total_behavior_num=200
    id_embedding_dim=64
elif dataset_name == "tmall":
    total_behavior_num=1000
    id_embedding_dim=16
simple_embedding_dim=8
layers1=1
layers2=1
drop_out=0.2
device_id = 0
compressed=True
diff1=False
diff2=False
layer = 2
use_cos=True
compress_net = "mlp"
random_init_q=False
gradient_gather=True

